package leetcode.study.tree;

/**
 * Created on 2021/1/4 。
 *
 * @author wll 1978992154@qq.com
 */

import common.TreeNode;

/**
 * 判断一颗树是不是高度平衡二叉树 二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
 * 一棵二叉树是平衡二叉树，当且仅当其所有子树也都是平衡二叉树，
 * 因此可以使用递归的方式判断二叉树是不是平衡二叉树，递归的顺序可以是自顶向下或者自底向上。
 * link:https://leetcode-cn.com/problems/balanced-binary-tree/
 */
public class IsBalanced {
    public boolean isBalanced(TreeNode root) {
        if (root == null) {
            return true;
        } else {
            return Math.abs(height(root.left) - height(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
        }
    }

    public int height(TreeNode root) {
        if (root == null) {
            return 0;
        } else {
            return Math.max(height(root.left), height(root.right)) + 1;
        }
    }
}
